import numpy as np
from NeuronNetwork import NeuronNetwork
neuron_train_x = [[3, 3],[1, 1],[4, 3],[2, 2],[5, 2],[1, 2],[4, 4],[3, 1],[4, 2],[1, 3],]
neuron_train_y = [[0],[1],[0],[1],[0],[1],[0],[1],[0],[1]]
def test():
    NN = NeuronNetwork(2, 4, 1)#隐藏层的维度 层数
    NN.train(
        neuron_train_x,
        neuron_train_y,
        eta=0.1,
        threshold=1e-7,
    )

test()